package com.free.space.common;

import java.io.File;
import java.io.FileInputStream;
import java.security.MessageDigest;
import java.util.HashMap;
import java.util.Map;

/**
 * MD5摘要工具类
 */
public class MD5DigestTool {
    static Map map = new HashMap<>();
    static int i=0;

    /**
     * Md5 摘要算法
     * @param key
     * @return
     */
    public static String MD5(String key) {
        char hexDigits[] = {
                '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'A', 'B', 'C', 'D', 'E', 'F'
        };
        try {
            byte[] btInput = key.getBytes();
            // 获得MD5摘要算法的 MessageDigest 对象
            MessageDigest mdInst = MessageDigest.getInstance("MD5");
            // 使用指定的字节更新摘要
            mdInst.update(btInput);
            // 获得密文
            byte[] md = mdInst.digest();
            // 把密文转换成十六进制的字符串形式
            int j = md.length;
            char str[] = new char[j * 2];
            int k = 0;
            for (int i = 0; i < j; i++) {
                byte byte0 = md[i];
                str[k++] = hexDigits[byte0 >>> 4 & 0xf];
                str[k++] = hexDigits[byte0 & 0xf];
            }
            return new String(str);
        } catch (Exception e) {
            return null;
        }
    }
    /**
     * 文件摘要算法
     * @param file
     */
    public static void md5DigestCompute(File file) {
        Long filelength = file.length();
        byte[] filecontent = new byte[filelength.intValue()];
        try {
            FileInputStream in = new FileInputStream(file);
            in.read(filecontent);
            in.close();
            String MD5_String = MD5(new String(filecontent));
            if(map.containsKey(MD5_String)){
                i++;
                System.err.println(i+ "文件 ： "+file.getName()+" 与 "+map.get(MD5_String)+" 重复 ");
            }else{
                map.put(MD5_String, file.getName());
            }
        } catch (Exception e) {
            System.err.println("The OS does not support ");
            e.printStackTrace();
        }
    }
}
